import time

from selenium.webdriver.common.by import By

from autoframe.common.base import Base


class IndexPage(Base):
    def search(self):                                             # 定义一个登录成功方法
        return self.get_element_text((By.CLASS_NAME, 'f4_b'))     # 通过class_name定位登录成功之后的页面

    def loginout(self):                                          # 定义一个退出登录的方法
        self.click((By.LINK_TEXT, '退出'))                        # 登录之后通过链接文本定位,点击
        time.sleep(1)